ExpandDims

对输入张量在给定的轴上添加额外维度。由于该算子仅改变张量形状，因此其DSP算子的作用是将数据从输入张量完整拷贝到输出张量。

输入：

  • src - 输入地址

  • total_copy_size - 计算得到的总共需拷贝的数据量，单位为字节。

  • core_mask - 核掩码。

输出：

  • dst - 输出地址。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持int8, int16, int32, fp32, fp64, cplx64, cplx128

  • MT7004 支持fp16, fp32, int16, int32, cplx64

共享/私有存储版本:

void anytype_expand_dims_anycore(void *src, void *dst, int total_copy_size, int core_mask)

各种数据类型、私有及共享空间版本均使用该函数。

C调用示例：

1void main(){
2    int core_mask = 0b1111; // 测试单核时核掩码设置为0b0001即可
3    int core_num = GetCoreNum(core_mask);
4    float* src = (float*)0x88000000; // 测试私有空间时地址设置在私有空间内即可
5    float* dst = (float*)0x98000000;
6    int shape[3] = {1, 10, 10};
7    int total_copy_size = shape[0] * shape[1] * shape[2] * sizeof(float);
8    anytype_expand_dims_anycore(src, dst, total_copy_size, core_mask);
9}